how to customize plots in python